<?php
namespace app\extend\WxUtil;
/**
 * 对微信小程序用户加密数据的解密示例代码.
 * @copyright Copyright (c) 1998-2014 Tencent Inc.
 */
class WxBizDataCrypt
{
    private $appid;
    private $sessionKey;

    /**
     * 构造函数
     * @param $sessionKey string 用户在小程序登录后获取的会话密钥
     * @param $appid string 小程序的appid
     */
    public function __construct($appid, $sessionKey)
    {
        $this->sessionKey = $sessionKey;
        $this->appid = $appid;
    }

    /**
     * 检验数据的真实性，并且获取解密后的明文.
     * @param $encryptedData string 加密的用户数据
     * @param $iv string 与用户数据一同返回的初始向量
     * @param $data string 解密后的原文
     *
     * @return int 成功0，失败返回对应的错误码
     */
    public function decryptData($encryptedData, $iv, &$data)
    {
        if (strlen($this->sessionKey) != 24) {
            return $this->ErrorCode('IllegalAesKey');
        }
        $aesKey = base64_decode($this->sessionKey);
        if (strlen($iv) != 24) {
            return $this->ErrorCode('IllegalIv');
        }
        $aesIV = base64_decode($iv);
        $aesCipher = base64_decode($encryptedData);
        $result = openssl_decrypt($aesCipher, "AES-128-CBC", $aesKey, 1, $aesIV);
        $dataObj = json_decode($result);
        if($dataObj == NULL) {
            return $this->ErrorCode('IllegalBuffer');
        }
        if($dataObj->watermark->appid != $this->appid) {
            return $this->ErrorCode('IllegalBuffer');
        }
        $data = $result;
        return $this->ErrorCode('OK');
    }

    /**
     * error code 说明.
     * <ul>
     *    <li>-41001: encodingAesKey 非法</li>
     *    <li>-41003: aes 解密失败</li>
     *    <li>-41004: 解密后得到的buffer非法</li>
     *    <li>-41005: base64加密失败</li>
     *    <li>-41016: base64解密失败</li>
     * </ul>
     */
    public function ErrorCode($name)
    {
        switch ($name) {
            case 'OK': return 0; break;
            case 'IllegalAesKey': return -41001; break; // 非法密钥
            case 'IllegalIv': return -41002; break; // 非法初始向量
            case 'IllegalBuffer': return -41003; break; // 非法密文
            case 'DecodeBase64Error': return -41004; break; // 解码错误
        }
    }
}
